package mine.code.question.动态规划;

import org.junit.Test;

import java.util.Arrays;

/**
 * 给定一个由 0 和 1 组成的矩阵 mat ，请输出一个大小相同的矩阵，其中每一个格子是 mat 中对应位置元素到最近的 0 的距离。
 * <p>
 * 两个相邻元素间的距离为 1 。
 *
 * @author caijinnan
 */
public class _01矩阵 {

    @Test
    public void run() {
        int[][] mat = {{0, 0, 0}, {0, 1, 0}, {1, 1, 1}};
        System.out.println(Arrays.deepToString(updateMatrix(mat)));
    }


    // 1,0,1,1
    // 1,1,1,1
    // 1,1,1,1
    // 1,1,1,0
    public int[][] updateMatrix(int[][] mat) {
        int xLength = mat.length;
        int yLength = mat[0].length;
        int[][] dp = new int[xLength][yLength];
        for (int i = 0; i < xLength; i++) {
            for (int j = 0; j < yLength; j++) {
                dp[i][j] = mat[i][j] == 0 ? 0 : Integer.MAX_VALUE / 2;
            }
        }

        for (int i = 0; i < xLength; ++i) {
            for (int j = 0; j < yLength; ++j) {
                if (i - 1 >= 0) {
                    dp[i][j] = Math.min(dp[i][j], dp[i - 1][j] + 1);
                }
                if (j - 1 >= 0) {
                    dp[i][j] = Math.min(dp[i][j], dp[i][j - 1] + 1);
                }
            }
        }
        // 只有 水平向右移动 和 竖直向下移动，注意动态规划的计算顺序
        for (int i = xLength - 1; i >= 0; --i) {
            for (int j = yLength - 1; j >= 0; --j) {
                if (i + 1 < xLength) {
                    dp[i][j] = Math.min(dp[i][j], dp[i + 1][j] + 1);
                }
                if (j + 1 < yLength) {
                    dp[i][j] = Math.min(dp[i][j], dp[i][j + 1] + 1);
                }
            }
        }
        return dp;
    }
}
